/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package vista.patologias;

import modelo.patologias.Enfermedad;
import modelo.patologias.TipoEnfermedad;
import util.components.MyTreeModel;

/**
 *
 * @author raul
 */
public class EnfermedadesTreeModelo extends MyTreeModel {

    @Override
    public Object getParent(Object node) {
        if (node instanceof Enfermedad){
            return ((Enfermedad) node).getTipoEnfermedad();
        } else if (node instanceof TipoEnfermedad) {
            return ((TipoEnfermedad) node).getTipo()!=null?((TipoEnfermedad) node).getTipo():getRoot();
        } else {
            return null;
        }
    }

    @Override
    public boolean isLeaf(Object node) {
        return node instanceof Enfermedad;
    }
    
}
